Divide Python list into multiple lists based on element condition